Python:shifting operation